import pandas as pd
import os


def main():
    # 读取Excel文件
    excel_file = '爱心募捐信息收集表（收集结果）.xlsx'
    df = pd.read_excel(excel_file)

    # 获取唯一的“党、团支部”列值
    unique_values = df['党、团支部'].unique()

    # 遍历每个唯一值，拆分表格并保存
    for value in unique_values:
        # 根据唯一值过滤数据
        filtered_df = df[df['党、团支部'] == value]

        # 创建存储拆分表格的文件夹
        group = os.path.join('output', value)  # 党、团支部列值
        workbook = group.split('/')[-1]  # 获取支部名称为表格文件名
        output_folder = group[:group.rfind('/')]  # 去除末尾支部名称并以之为文件夹路径
        os.makedirs(output_folder, exist_ok=True)  # 创建文件夹

        # 生成拆分后的Excel文件路径
        output_file = os.path.join(output_folder, f'{workbook}.xlsx')

        # 保存拆分后的表格
        filtered_df.to_excel(output_file, index=False)

    print("表格拆分完毕!")


if __name__ == '__main__':
    main()
